from selenium.webdriver.common.by import By
from testing.testHomework2.poHomeWork.Page.basePage import BasePage
from testing.testHomework2.poHomeWork.Page.contactPage import ContactPage


class MainPage(BasePage):
    _CONTACT = (By.ID, 'menu_contacts')#通讯录按钮
    #点击通讯录
    def getContact(self):
        self.find_and_click(*self._CONTACT)
        return ContactPage(self.driver)